"""工具节点模块 - 包含所有工具节点的定义。"""

from langgraph.prebuilt import ToolNode

from tools.search import search
from tools.user_profile import user_profile_finder
from tools.database import get_schema_tool, run_query_tool
from tools.sql_agent_tool import sql_query_tool

# React Agent 工具 - 添加SQL Agent工具
react_tools = [user_profile_finder, search, sql_query_tool]
react_tools_node = ToolNode(react_tools)

# SQL Agent 工具节点
get_schema_node = ToolNode([get_schema_tool])
run_query_node = ToolNode([run_query_tool])